//在一排树中，第 i 棵树产生 tree[i] 型的水果。 
//你可以从你选择的任何树开始，然后重复执行以下步骤： 
//
// 
// 把这棵树上的水果放进你的篮子里。如果你做不到，就停下来。 
// 移动到当前树右侧的下一棵树。如果右边没有树，就停下来。 
// 
//
// 请注意，在选择一颗树后，你没有任何选择：你必须执行步骤 1，然后执行步骤 2，然后返回步骤 1，然后执行步骤 2，依此类推，直至停止。 
//
// 你有两个篮子，每个篮子可以携带任何数量的水果，但你希望每个篮子只携带一种类型的水果。 
//
// 用这个程序你能收集的水果树的最大总量是多少？ 
//
// 
//
// 示例 1： 
//
// 输入：[1,2,1]
//输出：3
//解释：我们可以收集 [1,2,1]。
// 
//
// 示例 2： 
//
// 输入：[0,1,2,2]
//输出：3
//解释：我们可以收集 [1,2,2]
//如果我们从第一棵树开始，我们将只能收集到 [0, 1]。
// 
//
// 示例 3： 
//
// 输入：[1,2,3,2,2]
//输出：4
//解释：我们可以收集 [2,3,2,2]
//如果我们从第一棵树开始，我们将只能收集到 [1, 2]。
// 
//
// 示例 4： 
//
// 输入：[3,3,3,1,2,1,1,2,3,3,4]
//输出：5
//解释：我们可以收集 [1,2,1,1,2]
//如果我们从第一棵树或第八棵树开始，我们将只能收集到 4 棵水果树。
// 
//
// 
//
// 提示： 
//
// 
// 1 <= tree.length <= 40000 
// 0 <= tree[i] < tree.length 
// 
// Related Topics 数组 哈希表 滑动窗口 
// 👍 128 👎 0

package leetcode.editor.cn;

import java.util.HashMap;
import java.util.Map;

/**
 * [904]水果成篮
 */
public class FruitIntoBaskets {
    public static void main(String[] args) {
        Solution solution = new FruitIntoBaskets().new Solution();
        int total = 0;

//        int[] fruits = new int[]{1,2,1};
//        total = solution.totalFruit(fruits);
//        if (total != 3) {
//            System.out.println("--1-");
//        }
//
//        int[] fruits2 = new int[]{0,1,2,2};
//        total = solution.totalFruit(fruits2);
//        if (total != 3) {
//            System.out.println("--2-");
//        }
//
//        int[] fruits3 = new int[]{1,2,3,2,2};
//        total = solution.totalFruit(fruits3);
//        if (total != 3) {
//            System.out.println("--3-");
//        }

        int[] fruits4 = new int[]{3,3,3,1,2,1,1,2,3,3,4};
        total = solution.totalFruit(fruits4);
        if (total != 3) {
            System.out.println("--4-");
        }
    }
    //leetcode submit region begin(Prohibit modification and deletion)
class Solution {
    // 个人
    public int totalFruit2(int[] fruits) {
        int total = 0;
        int left = 0;
        Map<Integer, Integer> tree = new HashMap<>();
        Object object = new Object();
        for (int i = 0; i < fruits.length; i++) {
            int count = tree.getOrDefault(fruits[i], 0);
            tree.put(fruits[i], count+1);
            if (tree.size() <= 2) {
                total = Math.max(total, i - left + 1);
                continue;
            }

            while (tree.size() > 2) {
                count = tree.get(fruits[left]);
                if (count == 1) {
                    tree.remove(fruits[left]);
                } else {
                    tree.put(fruits[left], count-1);
                }
                left++;
            }
        }
        return total;
    }

    // 网友优化，滑动窗口不缩小
    public int totalFruit(int[] fruits) {
        Map<Integer, Integer> tree = new HashMap<>();
        int diff = 0;
        int left = 0;
        for (int i = 0; i < fruits.length; i++) {
            int count = tree.getOrDefault(fruits[i], 0);
            if (count == 0) {
                diff++;
            }
            tree.put(fruits[i], count+1);

            if (diff > 2) {
                count = tree.get(fruits[left]);
                if (count == 1) {
                    tree.remove(fruits[left]);
                    diff--;
                } else {
                    tree.put(fruits[left], count-1);
                }
                left++;
            }
        }
        return fruits.length - left;
    }
}
//leetcode submit region end(Prohibit modification and deletion)

}